package com.chuanyang.driverh.utils.widget;

import android.content.Context;
import android.content.res.TypedArray;
import android.util.AttributeSet;
import android.view.MotionEvent;
import android.view.View;
import android.view.ViewGroup;
import android.view.accessibility.AccessibilityEvent;
import android.view.accessibility.AccessibilityNodeInfo;
import android.widget.CompoundButton;
import android.widget.LinearLayout;
import android.widget.RadioButton;
import android.widget.RadioGroup;

/**
 * Created by xupeixian on 2017/3/14.
 */

public class MyRadioGroup extends LinearLayout {
        // holds the checked id; the selection is empty by default
        private int mCheckedId = -1;
        // tracks children radio buttons checked state
        private CompoundButton.OnCheckedChangeListener mChildOnCheckedChangeListener;
        // when true, mOnCheckedChangeListener discards events
        private boolean mProtectFromCheckedChange = false;
        private OnCheckedChangeListener mOnCheckedChangeListener;
        private PassThroughHierarchyChangeListener mPassThroughListener;

        /**
         * {@inheritDoc}
         */
        public MyRadioGroup(Context context) {
            super(context);
            setOrientation(VERTICAL);
            init();
        }

        /**
         * {@inheritDoc}
         */
        public MyRadioGroup(Context context, AttributeSet attrs) {
            super(context, attrs);
            mCheckedId = View.NO_ID;

            final int index = VERTICAL;
            setOrientation(index);

            init();
        }

        private void init() {
            mChildOnCheckedChangeListener = new CheckedStateTracker();
            mPassThroughListener = new PassThroughHierarchyChangeListener();
            super.setOnHierarchyChangeListener(mPassThroughListener);
        }

        /**
         * {@inheritDoc}
         */
        @Override
        public void setOnHierarchyChangeListener(OnHierarchyChangeListener listener) {
            // the user listener is delegated to our pass-through listener
            mPassThroughListener.mOnHierarchyChangeListener = listener;
        }

        /**
         * {@inheritDoc}
         */
        @Override
        protected void onFinishInflate() {
            super.onFinishInflate();

            // checks the appropriate radio button as requested in the XML file
            if (mCheckedId != -1) {
                mProtectFromCheckedChange = true;
                setCheckedStateForView(mCheckedId, true);
                mProtectFromCheckedChange = false;
                setCheckedId(mCheckedId);
            }
        }

        @Override
        public void addView(final View child, int index, ViewGroup.LayoutParams params) {
            if (child instanceof RadioButton) {

                ((RadioButton) child).setOnTouchListener(new OnTouchListener() {

                    @Override
                    public boolean onTouch(View v, MotionEvent event) {
                        ((RadioButton) child).setChecked(true);
                        checkRadioButton((RadioButton) child);
                        if(mOnCheckedChangeListener != null){
                            mOnCheckedChangeListener.onCheckedChanged(MyRadioGroup.this, child.getId());
                        }
                        return true;
                    }
                });

            } else if(child instanceof LinearLayout){
                int childCount = ((LinearLayout) child).getChildCount();
                for(int i = 0; i < childCount; i++){
                    View view = ((LinearLayout) child).getChildAt(i);
                    if (view instanceof RadioButton) {
                        final RadioButton button = (RadioButton) view;


                        ((RadioButton) button).setOnTouchListener(new OnTouchListener() {

                            @Override
                            public boolean onTouch(View v, MotionEvent event) {
                                ((RadioButton) button).setChecked(true);
                                checkRadioButton((RadioButton) button);
                                if(mOnCheckedChangeListener != null){
                                    mOnCheckedChangeListener.onCheckedChanged(MyRadioGroup.this, button.getId());
                                }
                                return true;
                            }
                        });

                    }
                }
            }

            super.addView(child, index, params);
        }

        private void checkRadioButton(RadioButton radioButton){
            View child;
            int radioCount = getChildCount();
            for(int i = 0; i < radioCount; i++){
                child = getChildAt(i);
                if (child instanceof RadioButton) {
                    if(child == radioButton){
                        // do nothing
                    } else {
                        ((RadioButton) child).setChecked(false);
                    }
                } else if(child instanceof LinearLayout){
                    int childCount = ((LinearLayout) child).getChildCount();
                    for(int j = 0; j < childCount; j++){
                        View view = ((LinearLayout) child).getChildAt(j);
                        if (view instanceof RadioButton) {
                            final RadioButton button = (RadioButton) view;
                            if(button == radioButton){
                                // do nothing
                            } else {
                                ((RadioButton) button).setChecked(false);
                            }
                        }
                    }
                }
            }
        }

        /**
         * <p>Sets the selection to the radio button whose identifier is passed in
         * parameter. Using -1 as the selection identifier clears the selection;
         * such an operation is equivalent to invoking {@link #clearCheck()}.</p>
         *
         * @param id the unique id of the radio button to select in this group
         *
         * @see #getCheckedRadioButtonId()
         * @see #clearCheck()
         */
        public void check(int id) {
            // don't even bother
            if (id != -1 && (id == mCheckedId)) {
                return;
            }

            if (mCheckedId != -1) {
                setCheckedStateForView(mCheckedId, false);
            }

            if (id != -1) {
                setCheckedStateForView(id, true);
            }

            setCheckedId(id);
        }

        private void setCheckedId(int id) {
            mCheckedId = id;
        }

        private void setCheckedStateForView(int viewId, boolean checked) {
            View checkedView = findViewById(viewId);
            if (checkedView != null && checkedView instanceof RadioButton) {
                ((RadioButton) checkedView).setChecked(checked);
            }
        }

        /**
         * <p>Returns the identifier of the selected radio button in this group.
         * Upon empty selection, the returned value is -1.</p>
         *
         * @return the unique id of the selected radio button in this group
         *
         * @see #check(int)
         * @see #clearCheck()
         *
         * @attr ref android.R.styleable#RadioGroup_checkedButton
         */
        public int getCheckedRadioButtonId() {
            return mCheckedId;
        }

        /**
         * <p>Clears the selection. When the selection is cleared, no radio button
         * in this group is selected and {@link #getCheckedRadioButtonId()} returns
         * null.</p>
         *
         * @see #check(int)
         * @see #getCheckedRadioButtonId()
         */
        public void clearCheck() {
            check(-1);
        }

        /**
         * <p>Register a callback to be invoked when the checked radio button
         * changes in this group.</p>
         *
         * @param listener the callback to call on checked state change
         */
        public void setOnCheckedChangeListener(OnCheckedChangeListener listener) {
            mOnCheckedChangeListener = listener;
        }

        /**
         * {@inheritDoc}
         */
        @Override
        public LayoutParams generateLayoutParams(AttributeSet attrs) {
            return new MyRadioGroup.LayoutParams(getContext(), attrs);
        }

        /**
         * {@inheritDoc}
         */
        @Override
        protected boolean checkLayoutParams(ViewGroup.LayoutParams p) {
            return p instanceof RadioGroup.LayoutParams;
        }

        @Override
        protected LinearLayout.LayoutParams generateDefaultLayoutParams() {
            return new LayoutParams(LayoutParams.WRAP_CONTENT, LayoutParams.WRAP_CONTENT);
        }

        @Override
        public void onInitializeAccessibilityEvent(AccessibilityEvent event) {
            super.onInitializeAccessibilityEvent(event);
            event.setClassName(RadioGroup.class.getName());
        }

        @Override
        public void onInitializeAccessibilityNodeInfo(AccessibilityNodeInfo info) {
            super.onInitializeAccessibilityNodeInfo(info);
            info.setClassName(RadioGroup.class.getName());
        }

        /**
         *
         */
        public static class LayoutParams extends LinearLayout.LayoutParams {
            /**
             * {@inheritDoc}
             */
            public LayoutParams(Context c, AttributeSet attrs) {
                super(c, attrs);
            }

            /**
             * {@inheritDoc}
             */
            public LayoutParams(int w, int h) {
                super(w, h);
            }

            /**
             * {@inheritDoc}
             */
            public LayoutParams(int w, int h, float initWeight) {
                super(w, h, initWeight);
            }

            /**
             * {@inheritDoc}
             */
            public LayoutParams(ViewGroup.LayoutParams p) {
                super(p);
            }

            /**
             * {@inheritDoc}
             */
            public LayoutParams(MarginLayoutParams source) {
                super(source);
            }

            /**
             * <p>Fixes the child's width to
             * {@link ViewGroup.LayoutParams#WRAP_CONTENT} and the child's
             * height to  {@link ViewGroup.LayoutParams#WRAP_CONTENT}
             * when not specified in the XML file.</p>
             *
             * @param a the styled attributes set
             * @param widthAttr the width attribute to fetch
             * @param heightAttr the height attribute to fetch
             */
            @Override
            protected void setBaseAttributes(TypedArray a,
                                             int widthAttr, int heightAttr) {

                if (a.hasValue(widthAttr)) {
                    width = a.getLayoutDimension(widthAttr, "layout_width");
                } else {
                    width = WRAP_CONTENT;
                }

                if (a.hasValue(heightAttr)) {
                    height = a.getLayoutDimension(heightAttr, "layout_height");
                } else {
                    height = WRAP_CONTENT;
                }
            }
        }

        /**
         * <p>Interface definition for a callback to be invoked when the checked
         * radio button changed in this group.</p>
         */
        public interface OnCheckedChangeListener {
            /**
             * <p>Called when the checked radio button has changed. When the
             * selection is cleared, checkedId is -1.</p>
             *
             * @param group the group in which the checked radio button has changed
             * @param checkedId the unique identifier of the newly checked radio button
             */
            public void onCheckedChanged(MyRadioGroup group, int checkedId);
        }

        private class CheckedStateTracker implements CompoundButton.OnCheckedChangeListener {
            public void onCheckedChanged(CompoundButton buttonView, boolean isChecked) {
                // prevents from infinite recursion
                if (mProtectFromCheckedChange) {
                    return;
                }

                mProtectFromCheckedChange = true;
                if (mCheckedId != -1) {
                    setCheckedStateForView(mCheckedId, false);
                }
                mProtectFromCheckedChange = false;

                int id = buttonView.getId();
                setCheckedId(id);
            }
        }

        /**
         * <p>A pass-through listener acts upon the events and dispatches them
         * to another listener. This allows the table layout to set its own internal
         * hierarchy change listener without preventing the user to setup his.</p>
         */
        private class PassThroughHierarchyChangeListener implements
                OnHierarchyChangeListener {
            private OnHierarchyChangeListener mOnHierarchyChangeListener;

            /**
             * {@inheritDoc}
             */
            public void onChildViewAdded(View parent, View child) {
                if (parent == MyRadioGroup.this && child instanceof RadioButton) {
                    int id = child.getId();
                    // generates an id if it's missing
                    if (id == View.NO_ID) {
                        id = child.hashCode();
                        child.setId(id);
                    }
                    ((RadioButton) child).setOnCheckedChangeListener(
                            mChildOnCheckedChangeListener);
                }

                if (mOnHierarchyChangeListener != null) {
                    mOnHierarchyChangeListener.onChildViewAdded(parent, child);
                }
            }

            /**
             * {@inheritDoc}
             */
            public void onChildViewRemoved(View parent, View child) {
                if (parent == MyRadioGroup.this && child instanceof RadioButton) {
                    ((RadioButton) child).setOnCheckedChangeListener(null);
                }

                if (mOnHierarchyChangeListener != null) {
                    mOnHierarchyChangeListener.onChildViewRemoved(parent, child);
                }
            }
        }
    }